import os

from mcp.server.fastmcp import FastMCP

from mcp_model import *
#定义一个MCP服务器
mcp = FastMCP("MCP Server")

@mcp.tool(name="add_tool",description="两个整数相加")
@authenticate()
def add_tool(x:int,y:int):
    """
    这个函数用于计算两个整数的和
    Args:
        x (int): 第一个整数
        y (int): 第二个整数
    Returns:
        int: 两个整数的和
    """
    token = os.getenv("Authorization")
    #写入日志
    with open("auth.log","a",encoding="utf-8") as f:
        f.write("token:"+token+"\n")
        f.write("===调用了add_tool ===\n")
    return x+y

@mcp.tool()
def sub_tool(x:int,y:int):
    """
    这个函数用于计算两个整数的差
    Args:
        x (int): 第一个整数
        y (int): 第二个整数
    Returns:
        int: 两个整数的差
    """
    print("===调用了sub_tool ===")
    return x-y


if __name__ == "__main__":
    print(" MCP Server 启动了...")
    #声明使用stdio传输协议,默认是sse协议
    mcp.run(transport="stdio")
    #mcp.run(transport="stdio")